package com.kzw.core.util;

import com.alibaba.fastjson.JSONObject;
import org.jdom.Attribute;
import org.jdom.Document;
import org.jdom.Element;
import org.jdom.input.SAXBuilder;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class XML2JSON {

	/** 
     * 转换一个xml格式的字符串到json格式 
     * @param xml  xml格式的字符串 
     * @return 成功返回json 格式的字符串;失败反回null 
     */  
    public static  String parse(String xml) {  
        JSONObject obj = new JSONObject();  
        try {  
            InputStream is = new ByteArrayInputStream(xml.getBytes("utf-8"));  
            SAXBuilder sb = new SAXBuilder();  
            Document doc = sb.build(is);  
            Element root = doc.getRootElement();  
            obj.put(root.getName(), iterateElement(root));  
            return obj.toString();  
        } catch (Exception e) {  
            e.printStackTrace();
        }  
        return null;  
    }  
  
    /** 
     * 转换一个xml格式的字符串到json格式 
     * @param file  java.io.File实例是一个有效的xml文件 
     * @return 成功反回json 格式的字符串;失败反回null 
     */  
    public static String parse(File file) {  
        JSONObject obj = new JSONObject();  
        try {  
            SAXBuilder sb = new SAXBuilder();  
            Document doc = sb.build(file);  
            Element root = doc.getRootElement();  
            obj.put(root.getName(), iterateElement(root));
            return obj.toString();  
        } catch (Exception e) {  
            e.printStackTrace();
        }  
        return null;  
    }  
  
    /** 
     * 一个迭代方法 
     */  
    @SuppressWarnings("unchecked")
	private static Map iterateElement(Element element) {
    	Map map = new HashMap();
    	//该元素的属性
    	List<Attribute> attrs = element.getAttributes();
    	for(Attribute attr : attrs) {
        	map.put(attr.getName(), attr.getValue());
        }
    	
    	//子元素
        List<Element> elms = element.getChildren();
        if(elms.size() > 0) {
        	List<Map> subList= new ArrayList<Map>();
        	for (int i=0; i<elms.size(); i++) {  
        		subList.add(iterateElement(elms.get(i)));
            }
        	map.put(elms.get(0).getName(), subList);
        }
        return map;
    }  
}
